package com.homework;

import java.util.ArrayList;
import java.util.Scanner;

public class StudentManager {
    //主页
    public static void main(String[] args) {
        //创建集合对象，用于存储学生数据
        ArrayList<Student> array = new ArrayList<Student>();
        //循环完成再次回到主页
        while (true) {
            //显示菜单

            //主界面
            System.out.println("-------欢迎来到学生管理系统-------");
            System.out.println("1.添加学生");
            System.out.println("2.删除学生");
            System.out.println("3.修改学生");
            System.out.println("4.查看所有学生");
            System.out.println("5.退出");
            System.out.println("请输入你的选择：");

            //键盘输入
            Scanner sc = new Scanner(System.in);
            String line = sc.nextLine();

            //使用switch语句完成操作选择
            switch (line) {
                case "1":
                    System.out.println("添加学生");
                    addStudent(array);
                    break;
                case "2":
                    System.out.println("删除学生");
                    deleteStudent(array);
                    break;
                case "3":
                    System.out.println("修改学生");
                    updateStudent(array);
                    break;
                case "4":
                    System.out.println("查看所有学生");
                    findAllStudent(array);
                    break;
                case "5":
                    System.out.println("谢谢使用");
                    System.exit(0);
            }
        }
    }

    //添加学生
    public static void addStudent(ArrayList<Student> array) {
        //键盘输入学生信息
        Scanner sc = new Scanner(System.in);

        System.out.println("请输入学生学号：");
        String id = sc.nextLine();
        System.out.println("请输入学生姓名：");
        String name = sc.nextLine();
        System.out.println("请输入学生年龄：");
        String age = sc.nextLine();
        System.out.println("请输入学生居住地：");
        String address = sc.nextLine();

        //创建学生对象，把键盘录入的数据赋值给学生对象的成员变量
        Student s = new Student();
        s.setId(id);
        s.setName(name);
        s.setAge(age);
        s.setAdress(address);

        //将学生对象添加到集合中
        array.add(s);

        //给出添加成功提示
        System.out.println("添加学生成功");

    }

    //查看所有学生
    public static void findAllStudent(ArrayList<Student> array) {
        //判断集合中是否有数据，如果没有显示提示信息
        if (array.size() == 0) {
            System.out.println("无信息，请先添加信息再查询");
            //不往下执行
            return;
        }

        //显示表头信息
        System.out.println("学号\t\t姓名\t\t年龄\t\t居住地");
        //取出集合中的数据按对应格式显示学生信息，年龄显示补充“岁”
        for (Student s : array) {
            System.out.println(s.getId() + "\t" + s.getName() + "\t" + s.getAge() + "岁\t" + s.getAdress());
        }
    }


    //建立一个方法，用于查找学生索引
    public static int getIndex(ArrayList<Student> array, String id) {
        int index = -1;
        for (int i = 0; i < array.size(); i++) {
            Student s = array.get(i);
            if (s.getId().equals(id)) {
                index = i;
                break;
            }
        }
        return index;
    }

    //删除学生
    public static void deleteStudent(ArrayList<Student> array) {
        //键盘录入要删除的学生学号，显示提示信息
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入你要删除的学生的学号：");
        String id = sc.nextLine();
        //遍历集合将对应学生对象从集合中删除
//        for (int i = 0; i < array.size(); i++) {
//            Student s = array.get(i);
//            if (s.getId().equals(id)) {
//                array.remove(i);
//                break;
//            }else {
//                System.out.println("查无此人");
//            }
//        }
        //删除/修改前先判断学号是否存在
//        int index = -1;
//        //获取索引
//        for (int i = 0; i < array.size(); i++) {
//            Student s = array.get(i);
//            if (s.getId().equals(id)) {
//                index = i;
//                break;
//            }
//        }

        int index = getIndex(array, id);

        if (index == -1) {
            System.out.println("查无此人");
        }else {
            array.remove(index);

        //给出删除成功提示
        System.out.println("删除学生成功");
        }

    }

    //修改学生
    public static void updateStudent(ArrayList<Student> array) {
        //键盘录入要修改的学生学号，显示提示信息
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入你要修改的学生的学号：");
        String id = sc.nextLine();
        //键盘录入要修改的学生信息
        System.out.println("请输入学生新姓名：");
        String name = sc.nextLine();
        System.out.println("请输入学生新年龄：");
        String age = sc.nextLine();
        System.out.println("请输入学生新居住地：");
        String address = sc.nextLine();
        //创建学生对象
        Student s = new Student();
        s.setId(id);
        s.setName(name);
        s.setAge(age);
        s.setAdress(address);
        //遍历集合修改对应的学生信息
//        for (int i = 0; i < array.size(); i++) {
//            Student student = array.get(i);
//            if (student.getId().equals(id)) {
//                array.set(i, s);
//                break;
//            }
//        }
        int index = getIndex(array, id);
        if (index == -1) {
            System.out.println("查无此人");
        }else {
            array.set(index, s);
        //给出修改成功提示
        System.out.println("修改学生成功");
        }
    }
}
